from ament_index_python.packages import get_package_share_directory
PATH_SHARE = get_package_share_directory('gui')
from pathlib import Path
PATH_ROOT = Path(PATH_SHARE).parents[3]
PATH_MODELS_DIR = PATH_ROOT / 'src/gui/detection_gui/scripts/models'
DEFECT_MODEL_PATH = PATH_MODELS_DIR / 'defect_detection_v2_20251103.pt'

DEFECT_COLORS = {
    'screw': (211, 193, 62),
    'wire': (0, 255, 0),
    'spot': (93, 22, 255)
}